package com.tanhua.sso.service;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.tanhua.sso.mapper.UserMapper;
import com.tanhua.sso.pojo.User;
import io.jsonwebtoken.ExpiredJwtException;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.messaging.MessagingException;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
 * @Author YuShiLong
 * @Date 2021/4/12 19:18
 * @Version 1.0
 */
@Service
@Slf4j
public class UserService {

    @Autowired
    private UserMapper userMapper;

    @Value("${jwt.secret}")
    private String secret;

    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    @Autowired
    private RocketMQTemplate rocketMQTemplate;

    @Autowired
    private HuanXinService huanXinService;


    public Map<String, Object> login(String phone, String code) {
        //校验验证码在Redis里面是否存在
        String redisKey = "CHECK_CODE_" + phone;

        //获取对应的Redis验证码
        String redisCode = redisTemplate.opsForValue().get(redisKey);

        //当输入的验证码和Redis里面的验证码不相同或者Redis里面的验证码为空,只要满足一个条件,就是错误的
        if (!StringUtils.equals(code, redisCode) || StringUtils.isEmpty(redisCode)) {
            //验证码错误
            return null;
        }

        //定义一个布尔值用来判断该用户是否为新用户
        boolean isNew = false;

        //判断手机号是否为注册过
        LambdaQueryWrapper<User> wrapper = new LambdaQueryWrapper<User>();
        wrapper.eq(User::getMobile, phone);
        //调用userMapper的selectOne方法查询数据库里面是否有该用户
        User user = userMapper.selectOne(wrapper);

        //user不存在,为新用户,为新用户注册信息
        if (user == null) {
            user = new User();
            user.setMobile(phone);
            user.setPassword(DigestUtils.md5Hex("123456"));

            //注册新用户
            this.userMapper.insert(user);
            //将布尔值改成true
            isNew = true;

            //注册用户到环信平台
            huanXinService.register(user.getId());
        }

        //生成token
        Map<String, Object> claims = new HashMap<String, Object>();
        //id主键自增,所有这里可以获取新用户的id,然后将用户的id放入token中
        claims.put("id", user.getId());

        // 生成token
        String token = Jwts.builder()
                .setClaims(claims) //payload，存放数据的位置，不能放置敏感数据，如：密码等
                .signWith(SignatureAlgorithm.HS256, secret) //设置加密方法和加密盐
                .setExpiration(new DateTime().plusHours(12).toDate()) //设置过期时间，12小时后过期
                .compact();

        //定义一个map,用来项controller返回数据
        Map<String, Object> map = new HashMap<>();
        map.put("token", token);
        map.put("isNew", isNew);

        //mq发送消息
        try {
            //发送用户登录成功的消息
            Map<String, Object> msg = new HashMap<>();
            msg.put("id", user.getId());
            msg.put("date", System.currentTimeMillis());

            this.rocketMQTemplate.convertAndSend("tanhua-sso-login", msg);
        } catch (MessagingException e) {
            log.error("发送消息失败！", e);
        }

        return map;
    }

    public User queryUserToken(String token) {

        try {
            // 通过token解析数据
            Map<String, Object> body = Jwts.parser()
                    .setSigningKey(secret)
                    .parseClaimsJws(token)
                    .getBody();
            //根据解析出来数据,获取用户id
            User user = new User();
            user.setId(Long.valueOf(body.get("id").toString()));

            //需要返回user对象中的mobile,需要查询数据库获取mobile数据
            //若果每次都去查询数据库,必然会导致性能问题需要对用户的手机号进行缓存操作
            //数据缓存时，需要设置过期时间，过期时间要与token的时间一致
            //如果用户修改了手机号，需要同步修改redis中的数据

            //定义一个字符串,作为存入Redis的key
            String redisKey = "TANHUA_USER_MOBILE_" + user.getId();
            //判断redis中是否有这条数据
            if (redisTemplate.hasKey(redisKey)) {
                //如果缓存中存在,获取这个数据
                String mobile = redisTemplate.opsForValue().get(redisKey);
                //将手机号码设置进user
                user.setMobile(mobile);
            } else {
                //如果缓存中不存在,就去查询数据库
                User u = userMapper.selectById(user.getId());
                //将手机号码设置进user
                user.setMobile(u.getMobile());

                //将手机号写入redis中
                //在jwt中的过期时间的单位为：秒
                long timeout = Long.valueOf(body.get("exp").toString()) * 1000 - System.currentTimeMillis();
                redisTemplate.opsForValue().set(redisKey, u.getMobile(), timeout, TimeUnit.MILLISECONDS);
            }

            return user;

        } catch (ExpiredJwtException e) {
            log.info("token已经过期！ token = " + token);
        } catch (Exception e) {
            log.error("token不合法！ token = " + token, e);
        }
        return null;
    }

}

